python avoid nested if statements